package com.itheima.health.mobile.utils;
import java.util.Random;
/**
 * @author ：seanyang
 * @date ：Created in 2019/6/11
 * @description ：随机生成验证码工具类
 * @version: 1.0
 */

/**
 * 随机生成验证码工具类
 */
public class ValidateCodeUtils {
	/**
	 * 随机生成验证码
	 * @param length 长度为4位或者6位
	 * @return
	 */
	public static Integer generateValidateCode(int length){
		Integer code =null;
		if(length == 4){
			code = new Random().nextInt(9999);//生成随机数，最大为9999
			if(code < 1000){
				code = code + 1000;//保证随机数为4位数字
			}
		}else if(length == 6){
			code = new Random().nextInt(999999);//生成随机数，最大为999999
			if(code < 100000){
				code = code + 100000;//保证随机数为6位数字
			}
		}else{
			throw new RuntimeException("只能生成4位或6位数字验证码");
		}
		return code;
	}

	/**
	 * 随机生成指定长度字符串验证码
	 * @param length 长度
	 * @return
	 */
	public static String generateValidateCode4String(int length){
		Random rdm = new Random();
		String hash1 = Integer.toHexString(rdm.nextInt());
		String capstr = hash1.substring(0, length);
		return capstr;
	}
}